#!/usr/bin/env python
#
'''
@File
@Time   :2022/06/14
@Author : Jiayu
'''

import os


class args():
    root = ''
    # 'data_path': '/home/jiayu/Desktop/JiaYu_data/1219(分六组)/601-1219(select317)/134第一轮/cell/revised_label(blue_and_red)/',
    # dataPath =  '/home/jiayu/MyProject_2022/data/Dataset(612)'
    dataPath = '/home/imed/personal/jiayu/2_Data/CCM_3'
    resultPath = '/home/imed/personal/jiayu/3-Result/Model/'
    saveModelName = "822_New_NF"
    saveModelPath = os.path.join(resultPath, saveModelName)

    env = saveModelName
    ### 134 new_nerve
    # 'data_path' : '/home/jiayu/Desktop/JiaYu_data/1219(分六组)/601-1219(select317)/Only_nerve_317',                ### 300 new_nerve
    # 'data_path':'/home/jiayu/MyProject_2022/Total_Dataset/Corneal_nerve_data',  ## CCM_1
    # testPath = '/home/jiayu/Desktop/JiaYu_data/1219(分六组)/601-1219(select317)/134第一轮/1+1_134_DataSet/val'
    testPath = '/home/imed/personal/jiayu/2_Data/CCM_3/val'
    modelPath = '/home/imed/personal/jiayu/3-Result/Model/822_New_NF/cell252-D0.9005-J0.8352.pth'
    predPath = '/home/imed/personal/jiayu/3-Result/Pred/0822_New_NF'
    savePredName1 = "Pred_c/"
    savePredName2 = "Pred_n/"

    savePred_c = os.path.join(predPath, savePredName1)
    savePred_n = os.path.join(predPath, savePredName2)
    epochs = 800
    lr = 0.01
    save_freq = 50
    test_step = 1
    ckpt_path = ' checkpoint_our/'
    # model_name = '618new'
    batch_size = 12
    no_cuda = False
    n_gpu = 2
    temp = 2
    set_loss = "n_dw"
